#!/usr/bin/env python3
# -*- coding: utf-8 -*-

"""
生成测试PDF文件的脚本
"""

from fpdf import FPDF
import os


def create_sample_pdf(filename, title, pages):
    """
    创建一个多页PDF文件用于测试
    
    Args:
        filename (str): 输出文件名
        title (str): PDF标题
        pages (int): 页数
    """
    pdf = FPDF()
    pdf.set_auto_page_break(auto=True, margin=15)
    
    # 添加页面并设置字体
    for i in range(pages):
        pdf.add_page()
        # 使用内置字体，避免中文编码问题
        pdf.set_font("Helvetica", size=16)
        pdf.cell(200, 10, txt=title, ln=True, align='C')
        pdf.ln(10)
        
        # 添加一些英文文本内容
        pdf.set_font("Helvetica", size=12)
        for j in range(20):
            pdf.cell(200, 10, txt=f"This is page {i+1}, line {j+1}.", ln=True)
    
    # 保存PDF
    pdf.output(filename)
    print(f"Created test PDF: {filename}")


def main():
    """主函数"""
    # 创建测试目录
    test_dir = "test_files"
    if not os.path.exists(test_dir):
        os.makedirs(test_dir)
    
    # 创建几个测试PDF文件
    create_sample_pdf(os.path.join(test_dir, "sample1.pdf"), "Sample Document 1", 3)
    create_sample_pdf(os.path.join(test_dir, "sample2.pdf"), "Sample Document 2", 2)
    create_sample_pdf(os.path.join(test_dir, "sample3.pdf"), "Sample Document 3", 5)
    
    print("All test PDF files have been created!")


if __name__ == "__main__":
    main()